import torch

# 检查是否支持 CUDA
has_cuda = torch.cuda.is_available()
print(f"PyTorch 是否支持 CUDA：{has_cuda}")

if has_cuda:
    # 输出 CUDA 的设备信息
    print(f"CUDA 设备数量：{torch.cuda.device_count()}")
    # 输出当前设备的名称
    print(f"当前 CUDA 设备名称：{torch.cuda.get_device_name(0)}")
else:
    print("当前环境不支持 CUDA")